# -*- coding:utf-8 -*-
# 实现一个自动取款机的存取款模拟效果。要求有登陆和退出、查询余额、取钱，存钱等操作

# 用户信息
user_info = {
    'aaa':{'password':111,'余额':999},
    'bbb':{'password':222,'余额':888},
    'ccc':{'password':333,'余额':777}
}

while True:
    print("*"*12, 'ATM系统', '*'*12)
    print('请登录：')
    user = input('请输入您的账号：')
    if user in user_info:  # 输入的账号是否存在
        password = input('请输入密码：')
        if int(password) == int(user_info[user]['password']): # 密码是否正确
            while True:
                print('{0:1}{1:10}{2:10}'.format(' ', '1.查询余额', '2.取款'))
                print('{0:1}{1:10}{2:10}'.format(' ', '3.存钱', '4.退出系统'))
                key1 = input('请选择您要操作的业务：')
                key = int(key1)
                if key == 1:    # 查询业务
                    print('卡上余额是：', str(user_info[user]['余额']))
                elif key == 2:  # 取款业务
                    saves = input('请输入取款金额：')
                    if int(saves) > int(user_info[user]['余额']):
                        print('余额不足')
                    else:   # 先将字典中的‘余额’删除，再添加
                        saves = int(user_info[user]['余额']) - int(saves)
                        del user_info[user]['余额']
                        user_info[user]['余额'] = saves
                        print('取款成功。剩余余额为：' + str(user_info[user]['余额']))
                elif key == 3:  # 存钱业务
                    saves = input('需要存储的金额：')
                    saves += int(user_info[user]['余额']) + int(saves)
                    user_info[user]['余额'] = str(saves)
                    print('存款成功。存储余额是：' + str(user_info[user]['余额']))
                elif key == 4:  # 退出操作
                    print('欢迎下次光临！')
                    break
                else:
                    print('输入不合法!')
        else:
            print('密码错误！')
    else:
        print('账号错误！')


